from .q_learner import QLearner
from .multi_head_learner import MultiHeadLearner

REGISTRY = {}

REGISTRY["q_learner"] = QLearner
REGISTRY["multi_head_learner"] = MultiHeadLearner